Volume of cylinder in python